import streamlit as st
import time
import sys
import os

print(sys.path)
print(os.getcwd())

# 设置页面标题
st.title("chat2money智金问答")
st.text("""beta 1.1 2024/12/27 22:30
        更新：从向量库chroma检索
        """)

# 创建一个输入框用于用户输入
user_input = st.text_input("投资不决问智金：")


# 2024/12/27 23:00 包路径错误
from app.rag.rag import RagManager
from app.utils.util import get_qwen_models

llm, chat, embed = get_qwen_models()
rag = RagManager(host="localhost", port=8000, llm=llm, embed=embed)


# 创建一个按钮用于发送消息
if st.button("发送"):
    result = rag.get_result("武汉力源信息'新型 GPS 信号处理装置'的专利号是多少？")

    # 更新对话历史
    st.write(f"AI: {result}")
    

